import { defineStore } from "pinia";
import {
  type userType,
  store,
  router,
  resetRouter,
  routerArrays,
  storageLocal
} from "../utils";
import type { UserResult, RefreshTokenResult } from "@/api/user";
import { useMultiTagsStoreHook } from "./multiTags";
import { type DataInfo, setToken, removeToken, userKey } from "@/utils/auth";
import {
  doLogin,
  getSessionInfo,
  refreshToken
} from "@/views/domain/session/api/session";

export const useUserStore = defineStore("pure-user", {
  state: (): userType => ({
    // 头像
    avatar: storageLocal().getItem<DataInfo<number>>(userKey)?.avatar ?? "",
    // 用户名
    username: storageLocal().getItem<DataInfo<number>>(userKey)?.username ?? "",
    // 昵称
    nickname: storageLocal().getItem<DataInfo<number>>(userKey)?.nickname ?? "",
    // 页面级别权限
    roles: storageLocal().getItem<DataInfo<number>>(userKey)?.roles ?? [],
    // 按钮级别权限
    permissions:
      storageLocal().getItem<DataInfo<number>>(userKey)?.permissions ?? [],
    // 前端生成的验证码（按实际需求替换）
    code: "",
    // 判断登录页面显示哪个组件（0：登录（默认）、1：手机登录、2：二维码登录、3：注册、4：忘记密码）
    currentPage: 0,
    // 是否勾选了登录页的免登录
    isRemembered: false,
    // 登录页的免登录存储几天，默认7天
    loginDay: 7
  }),
  actions: {
    /** 存储头像 */
    SET_AVATAR(avatar: string) {
      this.avatar = avatar;
    },
    /** 存储用户名 */
    SET_USERNAME(username: string) {
      this.username = username;
    },
    /** 存储昵称 */
    SET_NICKNAME(nickname: string) {
      this.nickname = nickname;
    },
    /** 存储角色 */
    SET_ROLES(roles: Array<string>) {
      this.roles = roles;
    },
    /** 存储按钮级别权限 */
    SET_PERMS(permissions: Array<string>) {
      this.permissions = permissions;
    },
    /** 存储前端生成的验证码 */
    SET_VERIFYCODE(code: string) {
      this.code = code;
    },
    /** 存储登录页面显示哪个组件 */
    SET_CURRENTPAGE(value: number) {
      this.currentPage = value;
    },
    /** 存储是否勾选了登录页的免登录 */
    SET_ISREMEMBERED(bool: boolean) {
      this.isRemembered = bool;
    },
    /** 设置登录页的免登录存储几天 */
    SET_LOGINDAY(value: number) {
      this.loginDay = Number(value);
    },
    /** 登入 */
    async loginByUsername(data): Promise<UserResult> {
      if (!data.username || !data.password) {
        return Promise.reject("Error");
      }

      const loginRes = await doLogin(data);
      if (!loginRes.token || loginRes.token === "") {
        return Promise.reject("Error");
      }
      const info = {
        accessToken: loginRes.token,
        refreshToken: loginRes.token,
        expires: new Date(loginRes.expire),
        permissions: [],
        avatar: ""
      };

      setToken(info);

      // 获取会话信息
      const sessRes = await getSessionInfo();
      if (!sessRes.data) {
        return Promise.reject(sessRes.msg);
      }

      return {
        success: true,
        data: {
          ...info,
          username: sessRes.data.userName,
          nickname: sessRes.data.userName,
          roles: [sessRes.data.roleId]
        }
      };
    },
    /** 前端登出（不调用接口） */
    logOut() {
      this.username = "";
      this.roles = [];
      this.permissions = [];
      removeToken();
      useMultiTagsStoreHook().handleTags("equal", [...routerArrays]);
      resetRouter();
      router.push("/login");
    },

    /** 刷新`token` */
    async handRefreshToken(): Promise<RefreshTokenResult> {
      const { token, expire, msg } = await refreshToken();
      if (!token || token === "") {
        return Promise.reject(msg);
      }

      const info = {
        accessToken: token,
        refreshToken: token,
        expires: new Date(expire),
        permissions: [],
        avatar: ""
      };

      setToken(info);

      return {
        success: true,
        data: info
      };
    }
  }
});

export function useUserStoreHook() {
  return useUserStore(store);
}
